/*******************************************************************************
* file  : aes demo
* creat : apleilx
* data  : 2017/03/02
* @brief  	aes 是一种对称加密算法，加密解密主KEY相同
*			该算法以128bit/192bit/256bit为单元进行加密解密，密文和明文尺寸相同
*			本测试代码为128bit模式加密、解密验证
*			16字节KEY，16字节明文，16字节密文
*			堆栈：2KB
*******************************************************************************/
#include "common.h"
#include "mbedtls/aes.h"
#include "bsp_api.h"
#include "sys_init.h"

// 此数组为 key_code_sim 代码二进制值
// 作为固定键值，可以伪装为程序
const uint16_t ase_key_src[] = {
	0xf04f, 0x5000, 0x6801, 0x07ca,
	0xd500, 0xbf00, 0x6841, 0x078a,
	0xd500, 0xbf00, 0x6881, 0x074a,
	0xd500, 0xbf00, 0x68c0, 0x0701,
	0xd500, 0xbf00, 0x4770};

#define dat_buff_init()              \
	loop_n(1000) src[index] = index; \
	memset(enc, 0, 1000);            \
	memset(dec, 0, 1000);

/*******************************************************************************
* @brief  des_demo.
* \param[in] none
* \retval: none
*******************************************************************************/
void aes_demo(void)
{
	uu32 clk_start;
	uu8 *src, *enc, *dec, *iv;
	mbedtls_aes_context *aes_enc, *aes_dec;
	size_t offset;

	// 内存请求
	src = calloc(1, 1024);
	enc = calloc(1, 1024);
	dec = calloc(1, 1024);
	iv = calloc(1, 80);
	aes_enc = malloc(sizeof(mbedtls_aes_context));
	aes_dec = malloc(sizeof(mbedtls_aes_context));

	dbg_puts("\n===========================================================\n");
	dbg_print("AES 128 encode test: <128 bits> \n");

	// 初始化
	dat_buff_init();

	// 初始化结构
	mbedtls_aes_init(aes_enc);
	mbedtls_aes_init(aes_dec);

	// 生成 KEY
	clk_start = bsp_cpu_clks_get();
	mbedtls_aes_setkey_enc(aes_enc, (void *)ase_key_src, 128);
	mbedtls_aes_setkey_dec(aes_dec, (void *)ase_key_src, 128);
	dbg_print("key make time: %d us\n", bsp_clks_to_us(bsp_cpu_clks_get() - clk_start));

	/*------------------------------------------------------------------------*/
	// 加密 + 解密 	ECB			电子密码本模式
	clk_start = bsp_cpu_clks_get();
	mbedtls_aes_crypt_ecb(aes_enc, MBEDTLS_AES_ENCRYPT, src, enc);
	mbedtls_aes_crypt_ecb(aes_dec, MBEDTLS_AES_DECRYPT, enc, dec);
	dbg_print("AES ecb : enc + dec time: %d us\n", bsp_clks_to_us(bsp_cpu_clks_get() - clk_start));

	// 显示
	//dbg_u8_print("src: ", src, 16, 20);
	dbg_u8_print("enc: ", enc, 16, 20);
	dbg_u8_print("dec: ", dec, 16, 20);

	/*------------------------------------------------------------------------*/
	// 加密 + 解密 	CBC			密码分组连接模式	- 内部分组 IO长度相同
	// 加密解密需要相同的初始向量, 向量长度16B
	dat_buff_init();
	clk_start = bsp_cpu_clks_get();

	memset(iv, 0xFF, 16);
	mbedtls_aes_crypt_cbc(aes_enc, MBEDTLS_AES_ENCRYPT, 512, iv, src, enc);

	memset(iv, 0xFF, 16);
	mbedtls_aes_crypt_cbc(aes_dec, MBEDTLS_AES_DECRYPT, 512, iv, enc, dec);

	dbg_print("\nAES CBC 512 : enc + dec time: %d us\n", bsp_clks_to_us(bsp_cpu_clks_get() - clk_start));

	// 显示
	//dbg_u8_print("src: \n", src, 64, 32);
	dbg_u8_print("enc: \n", enc, 128, 32);
	dbg_u8_print("dec: \n", dec, 128, 32);

	/*------------------------------------------------------------------------*/
	// 加密 + 解密 	CFB8		密文反馈模式		- 内部分组 IO长度相同
	// 加密解密需要相同的初始向量, 向量长度16B，加密解密都使用加密器
	dat_buff_init();
	mbedtls_aes_setkey_enc(aes_enc, (void *)ase_key_src, 128);
	clk_start = bsp_cpu_clks_get();
	memset(iv, 0xFF, 16);
	mbedtls_aes_crypt_cfb8(aes_enc, MBEDTLS_AES_ENCRYPT, 60, iv, src, enc);
	memset(iv, 0xFF, 16);
	mbedtls_aes_crypt_cfb8(aes_enc, MBEDTLS_AES_DECRYPT, 60, iv, enc, dec);
	dbg_print("\nAES CFB8 60 : enc + dec time: %d us\n", bsp_clks_to_us(bsp_cpu_clks_get() - clk_start));

	// 显示
	//dbg_u8_print("src: \n", src, 64, 32);
	dbg_u8_print("enc: \n", enc, 64, 32);
	dbg_u8_print("dec: \n", dec, 64, 32);

	/*------------------------------------------------------------------------*/
	// 加密 + 解密 	CFB128		密文反馈模式		- 内部分组 IO长度相同
	// 加密解密需要相同的初始向量, 向量长度16B，加密解密都使用加密器
	dat_buff_init();
	offset = 8; // 0~15
	clk_start = bsp_cpu_clks_get();

	memset(iv, 0xFF, 16);
	mbedtls_aes_setkey_enc(aes_enc, (void *)ase_key_src, 128);
	mbedtls_aes_crypt_cfb128(aes_enc, MBEDTLS_AES_ENCRYPT, 512, &offset, iv, src, enc);

	memset(iv, 0xFF, 16);
	mbedtls_aes_setkey_enc(aes_enc, (void *)ase_key_src, 128);
	mbedtls_aes_crypt_cfb128(aes_enc, MBEDTLS_AES_DECRYPT, 512, &offset, iv, enc, dec);

	dbg_print("\nAES CFB128 512 : enc + dec time: %d us\n", bsp_clks_to_us(bsp_cpu_clks_get() - clk_start));

	// 显示
	//dbg_u8_print("src: \n", src, 64, 32);
	dbg_u8_print("enc: \n", enc, 64, 32);
	dbg_u8_print("dec: \n", dec, 64, 32);

	/*------------------------------------------------------------------------*/
	// 加密 + 解密 	OFB		输出反馈模式
	// 加密解密需要相同的初始向量, 向量长度16B，加密解密都使用加密器
	dat_buff_init();
	offset = 8; // 0~15
	clk_start = bsp_cpu_clks_get();

	memset(iv, 0xFF, 16);
	mbedtls_aes_setkey_enc(aes_enc, (void *)ase_key_src, 128);
	mbedtls_aes_crypt_ofb(aes_enc, 512, &offset, iv, src, enc);

	memset(iv, 0xFF, 16);
	mbedtls_aes_setkey_enc(aes_enc, (void *)ase_key_src, 128);
	mbedtls_aes_crypt_ofb(aes_enc, 512, &offset, iv, enc, dec);

	dbg_print("\nAES OFB 512 : enc + dec time: %d us\n", bsp_clks_to_us(bsp_cpu_clks_get() - clk_start));

	// 显示
	//dbg_u8_print("src: \n", src, 64, 32);
	dbg_u8_print("enc: \n", enc, 64, 32);
	dbg_u8_print("dec: \n", dec, 64, 32);

	// 释放，用于清除内存中的密匙，属于安全性代码
	mbedtls_aes_free(aes_enc);
	mbedtls_aes_free(aes_dec);

	// 释放内存
	free(src);
	free(enc);
	free(dec);
	free(iv);
	free(aes_enc);
	free(aes_dec);
}
